# -*- coding: utf-8 -*-
"""
-------------------------------------------------
# @Project  :pythonProject
# @File     :main
# @Date     :2021/8/26 17:40
# @Author   :zhumenglong
# @Software :PyCharm
-------------------------------------------------
"""

from violator_special import *
from wareHouse import *

from violator_shoot import *
from violator_fall import *
from violator_cash import *
from violator_counter import *
from smokefire import *
from face import *
from skelton import *

if __name__ == '__main__':
    print("请输入您要执行的脚本类型的序号：1:special, 2:cash, 3:shoot, 4:fall, 5:counter, 6:fire, 7:face, 8:skelton")
    type_script = int(input())
    while(1):
        if type_script == 1 or type_script == 2 or type_script == 3 or type_script == 4 or type_script == 5 or \
                type_script == 6 or type_script == 7 or type_script == 8:
            break
        else:
            print("您输入的数据有误！！！  请重新输入：")
            type_script = int(input())

    print("请输入您要执行的原图的文件路径：(注：要以/结尾，例如：  D:/work/file/)")
    origin_path = str(input())
    Index_path = origin_path+'index.txt'

    print("请输入您要输出的目标图的文件名：(注：文件名即可，例如： work,默认会在相对路径下创建)")
    goal_path = str(input())

    if type_script == 1:
        sp = Special(origin_path,Index_path,goal_path)
        sp.get_keypoint()
        print("执行完毕！")
    elif type_script == 2:
        sp = Cash(origin_path,Index_path,goal_path)
        sp.draw_frame()
        print("执行完毕！")
    elif type_script == 3:
        sp = Shoot(origin_path, Index_path, goal_path)
        sp.draw_frame()
        print("执行完毕！")
    elif type_script == 4:
        sp = Fall(origin_path, Index_path, goal_path)
        sp.get_keypoint1()
        print("执行完毕！")
    elif type_script == 5:
        sp = Counter(origin_path, Index_path, goal_path)
        sp.get_keypoint()
        print("执行完毕！")
    elif type_script == 6:
        sp = Fire(origin_path, Index_path, goal_path)
        sp.get_keypoint()
        print("执行完毕！")
    elif type_script == 7:
        sp = Face(origin_path, Index_path, goal_path)
        sp.get_keypoint()
        print("执行完毕！")
    elif type_script == 8:
        sp = Skelton(origin_path, Index_path, goal_path)
        sp.get_keypoint1()
        print("执行完毕！")

